HideMethodSolver.java

/**
 * Copyright 2013 
 * 
 * HAN University of Applied Sciences
 * Maik Diepenbroek
 * Wouter Konecny
 * Sjoerd van den Top
 * Teun van Vegchel
 * Niek Versteege
 *
 * See the file MIT-license.txt for copying permission.
 */


package nl.han.ica.core.issue.solver;

import nl.han.ica.core.Delta;
import nl.han.ica.core.Parameter;
import nl.han.ica.core.Solution;
import nl.han.ica.core.SourceFile;
import nl.han.ica.core.issue.Issue;
import nl.han.ica.core.issue.IssueSolver;
import nl.han.ica.core.issue.detector.HideMethodDetector;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jdt.core.dom.ASTNode;
import org.eclipse.jdt.core.dom.MethodDeclaration;
import org.eclipse.jdt.core.dom.Modifier;
import org.eclipse.jdt.core.dom.rewrite.ASTRewrite;
import org.eclipse.jface.text.BadLocationException;
import org.eclipse.jface.text.IDocument;
import org.eclipse.text.edits.MalformedTreeException;
import org.eclipse.text.edits.TextEdit;

import java.io.IOException;
import java.util.Map;
import nl.han.ica.core.util.ASTUtil;
import org.eclipse.jdt.core.dom.TypeDeclaration;

/**
 * Solver for Hide Method.
 */
public class HideMethodSolver extends IssueSolver {

    private Map<String, Parameter> defaultParameters;

    public HideMethodSolver() {
        super();
    }

    @Override
    public boolean canSolve(Issue issue) {
        return issue.getDetector() instanceof HideMethodDetector;
    }

    //TODO clean
    @Override
    protected Solution internalSolve(Issue issue, Map<String, Parameter> parameters) {
        ASTNode node = issue.getNodes().get(0);
        Solution solution = new Solution(issue, this, parameters);
        SourceFile sourceFile = (SourceFile) node.getRoot().getProperty(SourceFile.SOURCE_FILE_PROPERTY);

        IDocument document = null;
        try {
            document = sourceFile.toDocument();
        } catch (IOException e) {
            // Log
        }

        Delta delta = solution.createDelta(sourceFile);
        delta.setBefore(document.get());

        ASTRewrite rewrite = ASTRewrite.create(node.getAST());
        MethodDeclaration newMethodDeclaration = (MethodDeclaration) ASTNode.copySubtree(node.getAST(), node);
        
        if(node instanceof MethodDeclaration){
            int modifiers = newMethodDeclaration.getModifiers();
            int modifierLocation = getAnnotationsSize((MethodDeclaration) node);
            if (Modifier.isPublic(modifiers) || Modifier.isProtected(modifiers)) {
                newMethodDeclaration.modifiers().remove(modifierLocation);

            }
            newMethodDeclaration.modifiers().addAll(modifierLocation, newMethodDeclaration.getAST().newModifiers(Modifier.PRIVATE));
        }

        rewrite.replace(node, newMethodDeclaration, null);
        TextEdit textEdit = rewrite.rewriteAST(document, JavaCore.getOptions());

        try {
            textEdit.apply(document);
        } catch (MalformedTreeException | BadLocationException e) {
            // Log
        }

        delta.setAfter(document.get());

        return solution;
    }

    private int getAnnotationsSize(MethodDeclaration declaration){
        if(declaration.resolveBinding().getAnnotations() != null) {
            return declaration.resolveBinding().getAnnotations().length;
        }
        return 0;
    }
}